sudo docker run --rm -v `pwd`:/src "hzy46/mt_pytorch:v1.7.0" python /src/tester.py \
    --package pytorch \
    --version "1.7.0" \
    --type gpu